Fix normalization of predictions