You can move this inside the if statement above