summaryrefslogtreecommitdiff
path: root/generator/vertex.py
diff options
context:
space:
mode:
Diffstat (limited to 'generator/vertex.py')
-rw-r--r--generator/vertex.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/generator/vertex.py b/generator/vertex.py
index 1fa61cb..3998e6e 100644
--- a/generator/vertex.py
+++ b/generator/vertex.py
@@ -3,8 +3,8 @@ import numpy as np
from math import sin, cos, sqrt
import copy
-import pi0
-from event import Event, Subtrack
+import chroma.pi0 as pi0
+from chroma.event import Event, Subtrack
# generator parts for use with gun()