diff --git a/wa/__init__.py b/wa/__init__.py
index d49e133a..2b4c71c8 100644
--- a/wa/__init__.py
+++ b/wa/__init__.py
@@ -14,4 +14,4 @@ from wa.framework.plugin import Plugin, Parameter
 from wa.framework.processor import ResultProcessor
 from wa.framework.resource import (NO_ONE, JarFile, ApkFile, ReventFile, File,
                                    Executable)
-from wa.framework.workload import Workload, ApkUiautoWorkload, ReventWorkload
+from wa.framework.workload import Workload, ApkWorkload, ApkUiautoWorkload, ReventWorkload