diff --git a/keras_mdn_layer/__init__.py b/keras_mdn_layer/__init__.py
index 02a4d05..75d718b 100644
--- a/keras_mdn_layer/__init__.py
+++ b/keras_mdn_layer/__init__.py
@@ -8,7 +8,6 @@
 
 Provided under MIT License
 """
-from .version import __version__
 from tensorflow import keras
 from tensorflow.keras import backend as K
 from tensorflow.keras import layers
diff --git a/keras_mdn_layer/version.py b/keras_mdn_layer/version.py
deleted file mode 100644
index 908c0bb..0000000
--- a/keras_mdn_layer/version.py
+++ /dev/null
@@ -1 +0,0 @@
-__version__ = '0.4.3'
diff --git a/pyproject.toml b/pyproject.toml
index 7249a76..f9daf33 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "keras-mdn-layer"
-version = "0.4.3"
+version = "0.4.4"
 description = "An MDN Layer for Keras using TensorFlow's distributions module"
 authors = ["Charles Martin <cpm@charlesmartin.au>"]
 license = "MIT"