Fixing main ui