diff --git a/devlib/module/gem5stats.py b/devlib/module/gem5stats.py
index ba24a43..f919905 100644
--- a/devlib/module/gem5stats.py
+++ b/devlib/module/gem5stats.py
@@ -17,6 +17,7 @@
 from collections import defaultdict
 
 import devlib
+from devlib.exception import TargetError
 from devlib.module import Module
 from devlib.platform import Platform
 from devlib.platform.gem5 import Gem5SimulationPlatform