Fix class not found error