fixed issue with class not being found