diff --git a/exp/createsamples.py b/exp/createsamples.py
--- a/exp/createsamples.py
+++ b/exp/createsamples.py
@@ -9,8 +9,7 @@ import PIL.ImageDraw
 import PIL
 
 from diagram import createDiagram
-from color_sampler import DataFile
-
+from annotations import DataFile
 
 finalPosCount=5000
 empty=True