MUI fixes