diff --git a/library/src/wavernn.cpp b/library/src/wavernn.cpp
index 4c6e12b..cabef83 100644
--- a/library/src/wavernn.cpp
+++ b/library/src/wavernn.cpp
@@ -307,6 +307,8 @@ Stretch2dLayer *Stretch2dLayer::loadNext(FILE *fd)
     fread( &header, sizeof(Stretch2dLayer::Header), 1, fd);
     x_scale = header.x_scale;
     y_scale = header.y_scale;
+    
+    return this;
 }
 
 Matrixf Stretch2dLayer::apply(const Matrixf &x)