Fix prediction