Looks like you can move this before normalization