diff --git a/tensorflow-ops/tests/BuildTest.hs b/tensorflow-ops/tests/BuildTest.hs
index 7a02232..63cb10c 100644
--- a/tensorflow-ops/tests/BuildTest.hs
+++ b/tensorflow-ops/tests/BuildTest.hs
@@ -114,7 +114,8 @@ testInitializedVariableShape =
         vector <- build $ do
                   a <- initializedVariable (constant [1] [42 :: Float])
                   b <- initializedValue (identity a)
-                  return b
+                  c <- initializedVariable b
+                  return c
         result <- run vector
         liftIO $ [42] @=? (result :: V.Vector Float)