Move code to else case