Fix change layout screen