diff --git a/assets/shaders/terrain.frag b/assets/shaders/terrain.frag
index 6cc0007..5998114 100644
--- a/assets/shaders/terrain.frag
+++ b/assets/shaders/terrain.frag
@@ -19,10 +19,10 @@ layout (set = 0, binding = 0) uniform Matrices {
 layout (set = 0, binding = 1) uniform sampler2D heightmap;
 
 void main() {
-	vec3 light_pos = vec3(10.0*cos(time), 10.0, 0.0);
-	vec3 L = normalize(light_pos-pos);
-	float r = length(light_pos-pos);
+	//vec3 light_pos = vec3(10.0*cos(time), 10.0, 0.0);
+	vec3 L = normalize(cam_pos-pos);
+	float r = length(cam_pos-pos);
 	float t = clamp(dot(L, norm), 0.0, 1.0) * 20.0/(r*r);
 
-	FragColor = vec4(t, t/2.0, t*2.0, 1.0);
+	FragColor = vec4(t*norm, 1.0);
 }
\ No newline at end of file
diff --git a/assets/shaders/terrain.frag.spv b/assets/shaders/terrain.frag.spv
index 6e07929..e8baa62 100644
Binary files a/assets/shaders/terrain.frag.spv and b/assets/shaders/terrain.frag.spv differ