fixed the class not found error