diff --git a/src/generate_unconditional_samples.py b/src/generate_unconditional_samples.py
index 205a754..2485149 100755
--- a/src/generate_unconditional_samples.py
+++ b/src/generate_unconditional_samples.py
@@ -6,7 +6,7 @@ import os
 import numpy as np
 import tensorflow as tf
 
-from src import model, sample, encoder
+import model, sample, encoder
 
 def sample_model(
     model_name='117M',