1 files changed, 3 insertions, 2 deletions
diff --git a/intern/cycles/render/graph.h b/intern/cycles/render/graph.h
index 426522066b3..7943d0ed063 100644
--- a/intern/cycles/render/graph.h
+++ b/intern/cycles/render/graph.h
@@ -147,7 +147,7 @@ public:
 	/* Simplify settings used by artists to the ones which are simpler to
 	 * evaluate in the kernel but keep the final result unchanged.
 	 */
-	virtual void simplify_settings(Scene * /*scene*/) {};
+	virtual void simplify_settings(Scene * /*scene*/, Shader * /*shader*/) {};
 
 	virtual bool has_surface_emission() { return false; }
 	virtual bool has_surface_transparent() { return false; }
@@ -246,8 +246,9 @@ public:
 	bool finalized;
 	bool simplified;
 	string displacement_hash;
+	Shader *shader;
 
-	ShaderGraph();
+	ShaderGraph(Shader *shader);
 	~ShaderGraph();
 
 	ShaderNode *add(ShaderNode *node);