diff --git a/minerva_analysis/server/models/data_model.py b/minerva_analysis/server/models/data_model.py
index cac23b87d..91f737e26 100644
--- a/minerva_analysis/server/models/data_model.py
+++ b/minerva_analysis/server/models/data_model.py
@@ -142,7 +142,7 @@ def load_csv(datasource_name, numpy=False):
 
     if 'CellType' in df.columns:
         df = df.rename(columns={'CellType': 'phenotype'})
-    if config[datasource_name]['featureData'][0]['celltype'] is not None:
+    if 'celltype' in config[datasource_name]['featureData'][0]:
         df = df.rename(columns={config[datasource_name]['featureData'][0]['celltype']: 'phenotype'})
 
     if np.issubdtype(df['phenotype'].dtype, np.number) is False: