diff --git a/dlls/wined3d/shader.c b/dlls/wined3d/shader.c
index 60e817485b8..a5b148648ed 100644
--- a/dlls/wined3d/shader.c
+++ b/dlls/wined3d/shader.c
@@ -3390,6 +3390,8 @@ static HRESULT shader_set_function(struct wined3d_shader *shader,
         return WINED3DERR_INVALIDCALL;
     }
 
+    shader->load_local_constsF = shader->lconst_inf_or_nan;
+
     return WINED3D_OK;
 }
 
@@ -3787,8 +3789,6 @@ static HRESULT shader_init(struct wined3d_shader *shader, struct wined3d_device
         }
     }
 
-    shader->load_local_constsF = shader->lconst_inf_or_nan;
-
     return hr;
 }