diff --git a/km3net_testdata/__init__.py b/km3net_testdata/__init__.py
index 3efeac2c61424ffbbaaa425d1b7687e1073bf5da..43f4c47e400be14cfaf07b351e5319d159a7286b 100644
--- a/km3net_testdata/__init__.py
+++ b/km3net_testdata/__init__.py
@@ -1,17 +1,10 @@
 from __future__ import absolute_import
 
-import os
+import atexit
 from pkg_resources import get_distribution, DistributionNotFound
 
 import importlib_resources
 
-try:
-    version = get_distribution(__name__).version
-except DistributionNotFound:
-    version = "unknown version"
-__version__ = version
-
-
 try:
     from importlib_resources import as_file
 except ImportError:
@@ -22,7 +15,14 @@ try:
 except ImportError:
     from contextlib2 import ExitStack
 
-import atexit
+
+try:
+    version = get_distribution(__name__).version
+except DistributionNotFound:
+    version = "unknown version"
+__version__ = version
+
+
 LATEST_VERSIONS = {
     "detx": "detx/detx_v3.detx",
 }