diff --git a/src/Flowpipes/Flowpipe.jl b/src/Flowpipes/Flowpipe.jl
index 18da6adf6..04b1ed896 100644
--- a/src/Flowpipes/Flowpipe.jl
+++ b/src/Flowpipes/Flowpipe.jl
@@ -270,7 +270,7 @@ function LazySets.linear_map(M, fp::Flowpipe)
     return Flowpipe(out, fp.ext)
 end
 
-function LazySets.affine_map(M, b, fp::Flowpipe)
+function LazySets.affine_map(M, fp::Flowpipe, b)
     out = [reconstruct(R, affine_map(M, set(R), b)) for R in fp]
     return Flowpipe(out, fp.ext)
 end
diff --git a/src/Flowpipes/MixedFlowpipe.jl b/src/Flowpipes/MixedFlowpipe.jl
index fc539acab..24f3b9dbb 100644
--- a/src/Flowpipes/MixedFlowpipe.jl
+++ b/src/Flowpipes/MixedFlowpipe.jl
@@ -82,7 +82,7 @@ function LazySets.linear_map(M, fp::MixedFlowpipe)
     return MixedFlowpipe(out, fp.ext)
 end
 
-function LazySets.affine_map(M, b, fp::MixedFlowpipe)
-    out = [affine_map(M, b, F) for F in fp]
+function LazySets.affine_map(M, fp::MixedFlowpipe, b)
+    out = [affine_map(M, F, b) for F in fp]
     return MixedFlowpipe(out, fp.ext)
 end