diff --git a/brainles_preprocessing/registration/__init__.py b/brainles_preprocessing/registration/__init__.py
index 9eea05d..b24c7a7 100644
--- a/brainles_preprocessing/registration/__init__.py
+++ b/brainles_preprocessing/registration/__init__.py
@@ -2,6 +2,8 @@
 
 try:
     from .ANTs.ANTs import ANTsRegistrator
+
+    ANTSRegistrator = ANTsRegistrator
 except ImportError:
     warnings.warn(
         "ANTS package not found. If you want to use it, please install it using 'pip install antspyx'"