Fixing ui shell on preview mode.