diff --git a/combo/main.py b/combo/main.py
index db75cb76619c052ea6bb8b0bdb6cebf4a1ab21bd..73b3a26ef2bfaafdf328fbfbfabc8ca96e887eef 100755
--- a/combo/main.py
+++ b/combo/main.py
@@ -7,7 +7,7 @@ from itertools import chain
 import random
 from typing import Dict, Optional, Any, Tuple
 
-import numpy.random.bit_generator
+import numpy
 import torch
 from absl import app, flags
 import pytorch_lightning as pl