diff --git a/scalex/function.py b/scalex/function.py
index 10fa9ef..7602fe5 100755
--- a/scalex/function.py
+++ b/scalex/function.py
@@ -8,7 +8,7 @@
 
 """
 
-# from anndata import AnnData
+from anndata import AnnData
 # from typing import Union, List
 
 
@@ -392,4 +392,4 @@ def main():
     
         
 if __name__ == '__main__':
-    main()
\ No newline at end of file
+    main()