diff --git a/neuralpy/layers/activation_functions/softmax.py b/neuralpy/layers/activation_functions/softmax.py
index e16783d..755c7f9 100644
--- a/neuralpy/layers/activation_functions/softmax.py
+++ b/neuralpy/layers/activation_functions/softmax.py
@@ -29,7 +29,7 @@ def __init__(self, dim=None, name=None):
         """
         super().__init__(_Softmax, "Softmax", layer_name=name)
 
-        if not isinstance(dim, int):
+        if dim is not None and not isinstance(dim, int):
             raise ValueError("Please provide a valid dim")
 
         self.__dim = dim