Fix predict function