diff --git a/marlin/__init__.py b/marlin/__init__.py
index 8b88e8f..b5b7758 100644
--- a/marlin/__init__.py
+++ b/marlin/__init__.py
@@ -80,7 +80,7 @@ def __init__(self, infeatures, outfeatures, groupsize=-1):
         super().__init__()
         if groupsize not in [-1, 128]:
             raise ValueError('Only groupsize -1 and 128 are supported.')
-        if infeatures % 128 != 0 or outfeatures != 256 == 0:
+        if infeatures % 128 != 0 or outfeatures % 256 != 0:
             raise ValueError('`infeatures` must be divisible by 128 and `outfeatures` by 256.')
         if groupsize == -1:
             groupsize = infeatures