diff --git a/neuralpy/layers/normalization/batchnorm1d.py b/neuralpy/layers/normalization/batchnorm1d.py
index 266514e..069492a 100644
--- a/neuralpy/layers/normalization/batchnorm1d.py
+++ b/neuralpy/layers/normalization/batchnorm1d.py
@@ -58,7 +58,7 @@ def __init__(
                 automatically calculates a unique name for the layer
         """
         # Checking num_features field
-        if num_features is not None and not isinstance(num_features, int):
+        if num_features is None or not isinstance(num_features, int):
             raise ValueError("Please provide a valid num_features")
         # Checking eps field
         if not isinstance(eps, float):