diff --git a/direct/src/particles/ParticleFloorTest.py b/direct/src/particles/ParticleFloorTest.py
index 3ee1bce319..c586e8fa48 100755
--- a/direct/src/particles/ParticleFloorTest.py
+++ b/direct/src/particles/ParticleFloorTest.py
@@ -1,5 +1,5 @@
 
-from panda3d.core import NodePath
+from panda3d.core import NodePath, Vec3
 from panda3d.physics import LinearVectorForce
 from direct.particles import ParticleEffect
 from direct.particles import Particles