diff --git a/train.py b/train.py
index 4b8f1f7..915eef9 100644
--- a/train.py
+++ b/train.py
@@ -2,9 +2,9 @@ import argparse
 import glob
 import math
 import os
+import random
 import time
 from pathlib import Path
-from random import random
 
 import numpy as np
 import torch.distributed as dist