diff --git a/cbpi/extension/mashstep/__init__.py b/cbpi/extension/mashstep/__init__.py
index 972b6b9..ace4e01 100644
--- a/cbpi/extension/mashstep/__init__.py
+++ b/cbpi/extension/mashstep/__init__.py
@@ -1,7 +1,7 @@
 import asyncio
 
-from cbpi.api import *
-from cbpi.api.step import StepResult
+from cbpi.api import parameters, Property, action
+from cbpi.api.step import StepResult, CBPiStep
 from cbpi.api.timer import Timer