Fixes class not found error on Flow-Math

Might be useful to merge. Not sure if other people have the same issue. But anyways this will fix it :+1: