diff --git a/src/renderers/webgl/WebGLProgram.js b/src/renderers/webgl/WebGLProgram.js
index 426d67a3606541..9340eff0f221d4 100644
--- a/src/renderers/webgl/WebGLProgram.js
+++ b/src/renderers/webgl/WebGLProgram.js
@@ -310,6 +310,7 @@ function WebGLProgram( renderer, code, material, parameters ) {
 	} else {
 
 		prefixVertex = [
+			'#define IS_VERTEX',
 
 			'precision ' + parameters.precision + ' float;',
 			'precision ' + parameters.precision + ' int;',
@@ -418,6 +419,8 @@ function WebGLProgram( renderer, code, material, parameters ) {
 
 			customExtensions,
 
+			'#define IS_FRAGMENT',
+
 			'precision ' + parameters.precision + ' float;',
 			'precision ' + parameters.precision + ' int;',