diff --git a/visualCaseGen/custom_widget_types/case_creator.py b/visualCaseGen/custom_widget_types/case_creator.py
index 8205857..93a5bd3 100644
--- a/visualCaseGen/custom_widget_types/case_creator.py
+++ b/visualCaseGen/custom_widget_types/case_creator.py
@@ -292,6 +292,16 @@ def _update_modelgrid_aliases(self, custom_grid_path, ocn_grid, do_exec):
         )
         new_ocnice_grid.text = ocn_grid
 
+        # Add grid for data runoff if drof component is selected:
+        if cvars["COMP_ROF"].value == "drof":
+            drof_grid = cvars["COMP_ROF_OPTION"].value
+            new_drof_grid = SubElement(
+                new_resolution,
+                "grid",
+                attrib={"name": "rof"},
+            )
+            new_drof_grid.text = drof_grid
+
         if not do_exec:
             return