diff --git a/CHANGELOG.md b/CHANGELOG.md
index 7990062..3a486e3 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -3,6 +3,10 @@
 ### Changes
 - Moved `neptune-xgboost` package to `src` directory ([#12](https://github.com/neptune-ai/neptune-xgboost/pull/12))
 
+### Fixes
+- Fixed import issue for callback - now it is possible to import as `from neptune_xgboost import NeptuneCallback`
+  ([#14](https://github.com/neptune-ai/neptune-xgboost/pull/14))
+
 ## neptune-xgboost 0.10.0
 
 ### Changes
diff --git a/src/neptune_xgboost/__init__.py b/src/neptune_xgboost/__init__.py
index 45b4013..615ef87 100644
--- a/src/neptune_xgboost/__init__.py
+++ b/src/neptune_xgboost/__init__.py
@@ -14,7 +14,7 @@
 # limitations under the License.
 #
 
-from ._version import get_versions
-
-__version__ = get_versions()["version"]
-del get_versions
+from neptune_xgboost.impl import (
+    NeptuneCallback,
+    __version__,
+)
diff --git a/src/neptune_xgboost/impl/__init__.py b/src/neptune_xgboost/impl/__init__.py
index 0eb0820..40d3ccb 100644
--- a/src/neptune_xgboost/impl/__init__.py
+++ b/src/neptune_xgboost/impl/__init__.py
@@ -16,6 +16,7 @@
 
 __all__ = [
     "NeptuneCallback",
+    "__version__",
 ]
 
 import json
@@ -27,8 +28,6 @@
 import xgboost as xgb
 from matplotlib import image
 
-from neptune_xgboost import __version__
-
 try:
     # neptune-client=0.9.0+ package structure
     import neptune.new as neptune
@@ -44,6 +43,10 @@
         verify_type,
     )
 
+from neptune_xgboost._version import get_versions
+
+__version__ = get_versions()["version"]
+
 INTEGRATION_VERSION_KEY = "source_code/integrations/neptune-xgboost"