fix normalization bug