diff --git a/src/python/m5/ext/pystats/__init__.py b/src/python/m5/ext/pystats/__init__.py
index d8d2b07a60..ce51c77f33 100644
--- a/src/python/m5/ext/pystats/__init__.py
+++ b/src/python/m5/ext/pystats/__init__.py
@@ -24,11 +24,22 @@
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
-from .abstract_stat import *
-from .group import *
-from .jsonloader import *
-from .serializable_stat import *
-from .simstat import *
-from .statistic import *
-from .storagetype import *
-from .timeconversion import *
+from .abstract_stat import AbstractStat
+from .group import (
+    Group,
+    SimObjectGroup,
+    SimObjectVectorGroup,
+)
+from .jsonloader import JsonLoader
+from .serializable_stat import SerializableStat
+from .simstat import SimStat
+from .statistic import (
+    Distribution,
+    Scalar,
+    SparseHist,
+    Statistic,
+    Vector,
+    Vector2d,
+)
+from .storagetype import StorageType
+from .timeconversion import TimeConversion