1 files changed, 3 insertions, 3 deletions
diff --git a/view.py b/view.py
index 3958e05..e99c46e 100755
--- a/view.py
+++ b/view.py
@@ -4,7 +4,7 @@ import numpy as np
 import pygame
 from pygame.locals import *
 
-import layout
+import src
 from camera import *
 from geometry import *
 from transform import *
@@ -35,8 +35,8 @@ def view(geometry, name=''):
 
     print 'device %s' % autoinit.device.name()
 
-    source = open(layout.source + '/kernel.cu').read()
-    module = SourceModule(source, options=['-I' + layout.source], no_extern_c=True, cache_dir=False)
+    module = SourceModule(src.kernel, options=['-I' + src.dir],
+                          no_extern_c=True, cache_dir=False)
     texrefs = geometry.load(module)
     cuda_raytrace = module.get_function('ray_trace')
     cuda_rotate = module.get_function('rotate')