diff --git a/runhouse/__init__.py b/runhouse/__init__.py
index c4efd5fd31..01161b0bc1 100644
--- a/runhouse/__init__.py
+++ b/runhouse/__init__.py
@@ -13,6 +13,8 @@
     OnDemandCluster,
 )
 
+from runhouse.resources.images import Image, ImageSetupStepType
+
 # WARNING: Any built-in module that is imported here must be capitalized followed by all lowercase, or we will
 # will not find the module class when attempting to reconstruct it from a config.
 from runhouse.resources.module import Module, module