diff --git a/enginecustom/applicationclass.cpp b/enginecustom/applicationclass.cpp
index 17adf6f..7b5305d 100644
--- a/enginecustom/applicationclass.cpp
+++ b/enginecustom/applicationclass.cpp
@@ -727,8 +727,8 @@ bool ApplicationClass::RenderRefractionToTexture()
 	// Render the bath model using the refraction shader.
 	m_BathModel->Render(m_Direct3D->GetDeviceContext());
 
-	result = m_ShaderManager->RenderRefractionShader(m_Direct3D->GetDeviceContext(), m_BathModel->GetIndexCount(), worldMatrix, viewMatrix, projectionMatrix, m_BathModel->GetTexture(0),
-		m_Lights[0]->GetDirection(), m_Light->GetAmbientColor(), m_Lights[0]->GetDiffuseColor(), clipPlane);
+	result = m_ShaderManager->RenderRefractionShader(m_Direct3D->GetDeviceContext(), m_BathModel->GetIndexCount(), worldMatrix, viewMatrix, projectionMatrix, 
+		m_BathModel->GetTexture(0), m_Lights[0]->GetDirection(), m_Lights[0]->GetAmbientColor(), m_Lights[0]->GetDiffuseColor(), clipPlane);
 	if (!result)
 	{
 		return false;
diff --git a/enginecustom/enginecustom.vcxproj b/enginecustom/enginecustom.vcxproj
index 891cd6b..6a6f8fa 100644
--- a/enginecustom/enginecustom.vcxproj
+++ b/enginecustom/enginecustom.vcxproj
@@ -211,6 +211,7 @@
     <CopyFileToFolders Include="sprite04.tga" />
     <CopyFileToFolders Include="stone01.tga" />
     <Image Include="wall.tga" />
+    <Image Include="water01.tga" />
   </ItemGroup>
   <ItemGroup>
     <CopyFileToFolders Include="cube.txt" />
diff --git a/enginecustom/enginecustom.vcxproj.filters b/enginecustom/enginecustom.vcxproj.filters
index 7df3b11..a3c1ca3 100644
--- a/enginecustom/enginecustom.vcxproj.filters
+++ b/enginecustom/enginecustom.vcxproj.filters
@@ -325,6 +325,9 @@
     <Image Include="KhaoticIcon.ico">
       <Filter>Assets</Filter>
     </Image>
+    <Image Include="water01.tga">
+      <Filter>Assets</Filter>
+    </Image>
   </ItemGroup>
   <ItemGroup>
     <None Include="packages.config" />