Fixing class not found error