From d47184ee57045a5cf014751b703db30d2ad4b8d2 Mon Sep 17 00:00:00 2001
From: Sean Lilley <lilleyse@gmail.com>
Date: Fri, 23 Oct 2015 10:00:12 -0400
Subject: [PATCH] Small change

---
 Source/Renderer/ShaderSource.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Source/Renderer/ShaderSource.js b/Source/Renderer/ShaderSource.js
index f8abc251bd0d..5f5c2bad7d4b 100644
--- a/Source/Renderer/ShaderSource.js
+++ b/Source/Renderer/ShaderSource.js
@@ -330,7 +330,7 @@ define([
     }
 
     ShaderSource.createPickVertexShaderSource = function(vertexShaderSource) {
-        var renamedVS = vertexShaderSource.replace(/void\s+main\s*\(\s*(?:void)?\s*\)/g, 'void czm_old_main()');
+        var renamedVS = ShaderSource.replaceMain(vertexShaderSource, 'void czm_old_main()');
         var pickMain = 'attribute vec4 pickColor; \n' +
             'varying vec4 czm_pickColor; \n' +
             'void main() \n' +
@@ -343,7 +343,7 @@ define([
     };
 
     ShaderSource.createPickFragmentShaderSource = function(fragmentShaderSource, pickColorQualifier) {
-        var renamedFS = fragmentShaderSource.replace(/void\s+main\s*\(\s*(?:void)?\s*\)/g, 'void czm_old_main()');
+        var renamedFS = ShaderSource.replaceMain(fragmentShaderSource, 'void czm_old_main()');
         var pickMain = pickColorQualifier + ' vec4 czm_pickColor; \n' +
             'void main() \n' +
             '{ \n' +