diff --git a/msre_power_history_restart.py b/msre_power_history_restart.py
index 948f4f3..88dbd2f 100644
--- a/msre_power_history_restart.py
+++ b/msre_power_history_restart.py
@@ -11,6 +11,7 @@ import math
 import numpy as np
 import re
 from pathlib import Path
+import h5py
 
 def get_geom_level_from_res(depletion_path, index):