diff --git a/__init__.py b/__init__.py
index a25960d..dff84d7 100644
--- a/__init__.py
+++ b/__init__.py
@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .timm_image import TimmImage
-
+from timm_image import TimmImage
 
 def timm(**kwargs):
-    return TimmImage(kwargs)
+    return TimmImage(**kwargs)