RTL на TextInput работает на всех остальных рендерах

У меня странная проблема с RTL, где, казалось бы, все переворачивается правильно, за исключением TextInput, которые работают ~ 50% времени. Вот гифка, показывающая проблему, когда я переключаю язык с английского на иврит:

(щелкните, чтобы увеличить изображение)

Вот компонент темы, в который я завернул свое дерево.

import React, {useContext} from 'react'
import {create} from 'jss'
import jssRtl from 'jss-rtl'
import JssProvider from 'react-jss/lib/JssProvider'
import {
  createGenerateClassName,
  jssPreset,
  MuiThemeProvider,
  createMuiTheme
} from '@material-ui/core/styles'
import CssBaseline from '@material-ui/core/CssBaseline'
import {UserContext} from 'src/context/user.js'

const generateClassName = createGenerateClassName()
const plugins = [...jssPreset().plugins]
const dir = locale => {
  const base = locale.split('-') || []
  return ['he', 'ar'].includes(base[0]) ? 'rtl' : 'ltr'
}

const setDir = dir => {
  window.document &&
    window.document.getElementsByTagName('body')[0].setAttribute('dir', dir)
}

export default function Theme({children}) {
  const {locale} = useContext(UserContext)
  const direction = dir(locale)
  const theme = createMuiTheme({direction, useNextVariants: true})
  if (direction === 'rtl') plugins.push(jssRtl())
  setDir(direction)
  console.info('theme', {locale, direction})

  return (
    <MuiThemeProvider theme = {theme}>
      <CssBaseline />
      {direction === 'rtl' ? (
        <JssProvider
          jss = {create({plugins})}
          generateClassName = {generateClassName}
        >
          {children}
        </JssProvider>
      ) : (
        children
      )}
    </MuiThemeProvider>
  )
}