Attempted to fix class not loading