diff --git a/torchview/__init__.py b/torchview/__init__.py
index 0b2e6be..b722b3d 100644
--- a/torchview/__init__.py
+++ b/torchview/__init__.py
@@ -12,4 +12,4 @@
     'TensorNode',
     'RecorderTensor',
 )
-__version__ = "0.2.3"
+__version__ = "0.2.4"