diff --git a/src/Futhark/IR/SegOp.hs b/src/Futhark/IR/SegOp.hs
index ced2d7b13d..d929c7b9b5 100644
--- a/src/Futhark/IR/SegOp.hs
+++ b/src/Futhark/IR/SegOp.hs
@@ -999,8 +999,7 @@ instance
   where
   cheapOp _ = False
   safeOp _ = True
-  opDependencies op =
-    replicate (length . kernelBodyResult $ segBody op) (freeIn op)
+  opDependencies op = replicate (length (segOpType op)) (freeIn op)
 
 --- Simplification