fix class not found bug