You should move this into `if` IMO.