diff --git a/ProcessLib/HydroMechanics/CreateHydroMechanicsProcess.cpp b/ProcessLib/HydroMechanics/CreateHydroMechanicsProcess.cpp
index 40b1264aec3d8f197404564f947dd576d9fa1213..c381578582e18576612e92a9bada8b56b4a0532d 100644
--- a/ProcessLib/HydroMechanics/CreateHydroMechanicsProcess.cpp
+++ b/ProcessLib/HydroMechanics/CreateHydroMechanicsProcess.cpp
@@ -143,13 +143,16 @@ std::unique_ptr<Process> createHydroMechanicsProcess(
         MaterialPropertyLib::porosity, MaterialPropertyLib::biot_coefficient,
         MaterialPropertyLib::density};
 
-    for (auto const& m : media)
+    for (auto const& element : mesh.getElements())
     {
-        checkRequiredProperties(*m.second, requiredMediumProperties);
-        checkRequiredProperties(m.second->phase("Gas"), requiredGasProperties);
-        checkRequiredProperties(m.second->phase("Solid"),
-                                requiredSolidProperties);
+        auto const element_id = element->getID();
+        media_map->checkElementHasMedium(element_id);
+        auto const& medium = *media_map->getMedium(element_id);
+        checkRequiredProperties(medium, requiredMediumProperties);
+        checkRequiredProperties(medium.phase("Gas"), requiredGasProperties);
+        checkRequiredProperties(medium.phase("Solid"), requiredSolidProperties);
     }
+    DBUG("Media properties verified.");
 
     // Initial stress conditions
     auto const initial_stress = ParameterLib::findOptionalTagParameter<double>(