Fixing feature normalization