diff --git a/ivy/utils/decorator_utils.py b/ivy/utils/decorator_utils.py
index b36bee9244d9..fe7bc0755b74 100644
--- a/ivy/utils/decorator_utils.py
+++ b/ivy/utils/decorator_utils.py
@@ -432,8 +432,11 @@ def transpose_wrapper(*args, **kwargs):
                 kwargs["data_format"] = value_map[kwargs["data_format"]]
             if "filter_format" in kwargs and kwargs["filter_format"] in value_map:
                 kwargs["filter_format"] = value_map[kwargs["filter_format"]]
+                os.environ["DATA_FORMAT"] = "channels_last"
 
-        return fn(*args, **kwargs)
+        res = fn(*args, **kwargs)
+        os.environ["DATA_FORMAT"] = DATA_FORMAT
+        return res
 
     return transpose_wrapper