Fix screen after registration