1 files changed, 6 insertions, 6 deletions
diff --git a/lib/mkmf.rb b/lib/mkmf.rb
index ce88760230..c5cfad30bb 100644
--- a/lib/mkmf.rb
+++ b/lib/mkmf.rb
@@ -294,25 +294,25 @@ end
 def with_cppflags(flags)
   cppflags = $CPPFLAGS
   $CPPFLAGS = flags
-  return yield
+  ret = yield
 ensure
-  $CPPFLAGS = cppflags
+  $CPPFLAGS = cppflags unless ret
 end
 
 def with_cflags(flags)
   cflags = $CFLAGS
   $CFLAGS = flags
-  return yield
+  ret = yield
 ensure
-  $CFLAGS = cflags
+  $CFLAGS = cflags unless ret
 end
 
 def with_ldflags(flags)
   ldflags = $LDFLAGS
   $LDFLAGS = flags
-  return yield
+  ret = yield
 ensure
-  $LDFLAGS = ldflags
+  $LDFLAGS = ldflags unless ret
 end
 
 def try_static_assert(expr, headers = nil, opt = "", &b)