diff --git a/src/maggotuba/models/modules.py b/src/maggotuba/models/modules.py
index f567e736e4054da91ba5aac825e8e82aa911519d..bc3aada8fb845b0254ae803f2f32ae972ca01e47 100644
--- a/src/maggotuba/models/modules.py
+++ b/src/maggotuba/models/modules.py
@@ -1,6 +1,7 @@
 import logging
 import os
 from pathlib import Path
+import numpy as np
 import torch
 from torch import nn
 import json