Fix prediction dialog