diff --git a/metaflow/plugins/__init__.py b/metaflow/plugins/__init__.py
index 00a5bf2d106..9a18fad8bfc 100644
--- a/metaflow/plugins/__init__.py
+++ b/metaflow/plugins/__init__.py
@@ -161,7 +161,6 @@ def get_plugin_cli():
 from .aws.step_functions.schedule_decorator import ScheduleDecorator
 from .project_decorator import ProjectDecorator
 from .aip.s3_sensor_decorator import S3SensorDecorator
-
 from .aip.exit_handler_decorator import ExitHandlerDecorator
 
 FLOW_DECORATORS = [
diff --git a/metaflow/plugins/aip/tests/flows/skip_flow.py b/metaflow/plugins/aip/tests/flows/skip_flow.py
index 23ae936f55a..0b2eb3aa078 100644
--- a/metaflow/plugins/aip/tests/flows/skip_flow.py
+++ b/metaflow/plugins/aip/tests/flows/skip_flow.py
@@ -18,13 +18,14 @@ def skipped_step(self):
         raise Exception(
             "Unexpectedly ran the skipped_step step. This step should have been skipped."
         )
-        self.next(self.unreachable)
+        self.next(self.unreachable_step)
 
-    def unreachable(self):
+    @step
+    def unreachable_step(self):
         raise Exception(
             "Unexpectedly ran the unreachable step. This step should have been skipped."
         )
-        self.next(self.end)
+        self.next(self.desired_step)
 
     @skip(check="condition_false", next="end")
     @step