Fix normalization bug