--- a/trafficintelligence/tests/prediction.txt	Mon Jun 18 22:50:59 2018 -0400
+++ b/trafficintelligence/tests/prediction.txt	Tue Jun 19 10:04:52 2018 -0400
@@ -1,6 +1,6 @@
->>> from prediction import *
->>> import moving, storage, utils
->>> from numpy import absolute, array
+>>> from trafficintelligence.prediction import *
+>>> from trafficintelligence import moving, storage, utils
+>>> from numpy import absolute, array, max
 
 >>> et = PredictedTrajectoryConstant(moving.Point(0,0), moving.Point(1,0))
 >>> et.predictPosition(4) # doctest:+ELLIPSIS
@@ -21,7 +21,6 @@
 >>> steering = lambda: random.uniform(-0.1,0.1)
 >>> et = PredictedTrajectoryRandomControl(moving.Point(0,0),moving.Point(1,1), acceleration, steering, maxSpeed = 2)
 >>> p = et.predictPosition(500)
->>> from numpy import max
 >>> max(et.getPredictedSpeeds()) <= 2.
 True