diff --git a/FDTD/operator.cpp b/FDTD/operator.cpp
index 463a3fe..a7582aa 100644
--- a/FDTD/operator.cpp
+++ b/FDTD/operator.cpp
@@ -721,7 +721,7 @@ void Operator::DumpPEC2File(string filename , unsigned int *range)
 	filename += ".vtp";
 	writer->SetFileName(filename.c_str());
 
-#if VTK_MAJOR_VERSION==6
+#if VTK_MAJOR_VERSION>=6
 	writer->SetInputData(polydata);
 #else
 	writer->SetInput(polydata);
diff --git a/tools/vtk_file_writer.cpp b/tools/vtk_file_writer.cpp
index b1fe3e7..79c40f3 100644
--- a/tools/vtk_file_writer.cpp
+++ b/tools/vtk_file_writer.cpp
@@ -285,7 +285,7 @@ bool VTK_File_Writer::WriteASCII()
 	}
 
 	writer->SetHeader(m_header.c_str());
-#if VTK_MAJOR_VERSION==6
+#if VTK_MAJOR_VERSION>=6
 	writer->SetInputData(m_GridData);
 #else
 	writer->SetInput(m_GridData);
@@ -316,7 +316,7 @@ bool VTK_File_Writer::WriteXML()
 		return false;
 	}
 
-#if VTK_MAJOR_VERSION==6
+#if VTK_MAJOR_VERSION>=6
 	writer->SetInputData(m_GridData);
 #else
 	writer->SetInput(m_GridData);