diff --git a/pytRIBS/mesh/mesh.py b/pytRIBS/mesh/mesh.py
index 4288053..394039d 100644
--- a/pytRIBS/mesh/mesh.py
+++ b/pytRIBS/mesh/mesh.py
@@ -598,6 +598,7 @@ def extract_watershed_and_stream_network(self, outlet_path, boundary_path, outpu
 
         if clean is True:
             shutil.rmtree(temp)
+            self.output_dir = output_dir
 
 
 class GenerateMesh: