diff --git a/setup.py b/setup.py
index c3f6e27..de78f37 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'siren-pytorch',
   packages = find_packages(),
-  version = '0.1.4',
+  version = '0.1.5',
   license='MIT',
   description = 'Implicit Neural Representations with Periodic Activation Functions',
   author = 'Phil Wang',
diff --git a/siren_pytorch/siren_pytorch.py b/siren_pytorch/siren_pytorch.py
index 910c887..ed059aa 100644
--- a/siren_pytorch/siren_pytorch.py
+++ b/siren_pytorch/siren_pytorch.py
@@ -83,7 +83,7 @@ def forward(self, x, mods = None):
             x = layer(x)
 
             if exists(mod):
-                x *= rearrange(mod, 'd -> () d').sigmoid()
+                x *= rearrange(mod, 'd -> () d')
 
         return self.last_layer(x)