diff --git a/nvcc4jupyter/plugin.py b/nvcc4jupyter/plugin.py
index f5b77d4..e278d56 100644
--- a/nvcc4jupyter/plugin.py
+++ b/nvcc4jupyter/plugin.py
@@ -301,3 +301,8 @@ class NVCCPlugin(Magics):
             return
 
         self._delete_group(args.group)
+
+
+def load_ipython_extension(shell: InteractiveShell):
+    nvcc_plugin = NVCCPlugin(shell)
+    shell.register_magics(nvcc_plugin)
diff --git a/nvcc_plugin.py b/nvcc_plugin.py
deleted file mode 100644
index c5afcfe..0000000
--- a/nvcc_plugin.py
+++ /dev/null
@@ -1,8 +0,0 @@
-from IPython.core.interactiveshell import InteractiveShell
-
-from v1.v1 import NVCCPlugin as NVCC_V1
-
-
-def load_ipython_extension(shell: InteractiveShell):
-    nvcc_plugin = NVCC_V1(shell)
-    shell.register_magics(nvcc_plugin)