Fixing prediction again..oy