some fixes to make predictions