Fixes class not found error on Flow-Math