diff --git a/bigbio/hub/hub_repos/czi_drsm/README.md b/bigbio/hub/hub_repos/czi_drsm/README.md
index 7b7e6ee6..2d950e6e 100644
--- a/bigbio/hub/hub_repos/czi_drsm/README.md
+++ b/bigbio/hub/hub_repos/czi_drsm/README.md
@@ -3,7 +3,7 @@ language:
   - en 
 bigbio_language:
   - English
-license: Creative Commons Zero v1.0 Universal
+license: cc0-1.0
 bigbio_license_shortname: cc0-1.0
 multilinguality: monolingual
 pretty_name: CZI DRSM
diff --git a/bigbio/hub/hub_repos/czi_drsm/czi_drsm.py b/bigbio/hub/hub_repos/czi_drsm/czi_drsm.py
index 24f54d45..68e3ed3b 100644
--- a/bigbio/hub/hub_repos/czi_drsm/czi_drsm.py
+++ b/bigbio/hub/hub_repos/czi_drsm/czi_drsm.py
@@ -60,16 +60,12 @@
 import pandas as pd
 from pathlib import Path
 
-import bigbio.utils.parsing as parse
-from bigbio.utils import schemas
-from bigbio.utils.configs import BigBioConfig
-from bigbio.utils.constants import Lang, Tasks
-from bigbio.utils.license import Licenses
+from .bigbiohub import text_features, BigBioConfig, Tasks
 
 #from .bigbiohub import BigBioConfig
 #from .bigbiohub import Tasks
 
-#from .bigbiohub import 
+#from .bigbiohub import
 
 _LOCAL = False
 
@@ -147,7 +143,7 @@
     'nhs': "https://raw.githubusercontent.com/chanzuckerberg/DRSM-corpus/main/v2/nhs_all_2023_03_31.tsv"
 }
 
-_SUPPORTED_TASKS = [Tasks.TEXT_CLASSIFICATION]  
+_SUPPORTED_TASKS = [Tasks.TEXT_CLASSIFICATION]
 
 _SOURCE_VERSION = "1.0.0"
 _BIGBIO_VERSION = "1.0.0"
@@ -295,7 +291,7 @@ def _info(self) -> datasets.DatasetInfo:
 
         # For example bigbio_kb, bigbio_t2t
         elif self.config.schema == "bigbio_text":
-            features = schemas.text_features
+            features = text_features
 
         return datasets.DatasetInfo(
             description=_DESCRIPTION,