diff --git a/src/main.cc b/src/main.cc
index 3d30b6a..e6b7af2 100644
--- a/src/main.cc
+++ b/src/main.cc
@@ -55,12 +55,16 @@ int main(int argc, char **argv) {
 	removeExternalCellsFilter->SetInputConnection(1,
 		polyMeshReader->GetOutputPort());
 
-	// vtkNew<ProjectSurfacePointsOnPoly> project;
-	// project->SetInputConnection(0, removeExternalCellsFilter->GetOutputPort());
-	// project->SetInputConnection(1, pdReader->GetOutputPort());
+	vtkNew<SurfacePointsFilter> surfacePointsFilter;
+	surfacePointsFilter->SetInputConnection(
+		removeExternalCellsFilter->GetOutputPort());
+
+	vtkNew<ProjectSurfacePointsOnPoly> project;
+	project->SetInputConnection(0, surfacePointsFilter->GetOutputPort());
+	project->SetInputConnection(1, polyMeshReader->GetOutputPort());
 
 	vtkNew<vtkXMLUnstructuredGridWriter> writer;
-	writer->SetInputConnection(removeExternalCellsFilter->GetOutputPort());
+	writer->SetInputConnection(project->GetOutputPort());
 	writer->SetDataModeToAscii();
 	writer->SetFileName("out.vtu");
 	writer->Write();