diff --git a/pts/modules/flows.py b/pts/modules/flows.py
index 582e8df..5f90f27 100644
--- a/pts/modules/flows.py
+++ b/pts/modules/flows.py
@@ -304,6 +304,7 @@ class MADE(nn.Module):
 
 class Flow(nn.Module):
     def __init__(self, input_size):
+        super().__init__()
         self.__scale = None
         self.net = None