Fixes prediction step