diff --git a/test/porousmediumflow/richards/nonisothermal/evaporation/properties.hh b/test/porousmediumflow/richards/nonisothermal/evaporation/properties.hh
index 1f1b341e169d2309ae74cde77e3f4abe002e5ea3..f7f34c5d1ee1f5986c111904fb07896f7507118a 100644
--- a/test/porousmediumflow/richards/nonisothermal/evaporation/properties.hh
+++ b/test/porousmediumflow/richards/nonisothermal/evaporation/properties.hh
@@ -32,7 +32,7 @@
 #include <dumux/discretization/cctpfa.hh>
 #include <dumux/discretization/box.hh>
 
-#include <dumux/porousmediumflow/richards/model.hh>
+#include <dumux/porousmediumflow/richardsextended/model.hh>
 #include <dumux/material/fluidmatrixinteractions/2p/thermalconductivity/somerton.hh>
 #include <dumux/material/fluidsystems/h2on2.hh>
 
@@ -43,7 +43,7 @@ namespace Dumux::Properties {
 
 // Create new type tags
 namespace TTag {
-struct RichardsNIEvaporation { using InheritsFrom = std::tuple<RichardsNI>; };
+struct RichardsNIEvaporation { using InheritsFrom = std::tuple<ExtendedRichardsNI>; };
 struct RichardsNIEvaporationBox { using InheritsFrom = std::tuple<RichardsNIEvaporation, BoxModel>; };
 struct RichardsNIEvaporationCC { using InheritsFrom = std::tuple<RichardsNIEvaporation, CCTpfaModel>; };
 } // end namespace TTag
@@ -69,9 +69,6 @@ struct SpatialParams<TypeTag, TTag::RichardsNIEvaporation>
     using type = RichardsNISpatialParams<GridGeometry, Scalar>;
 };
 
-template<class TypeTag>
-struct EnableWaterDiffusionInAir<TypeTag, TTag::RichardsNIEvaporation> { static constexpr bool value = true; };
-
 } // end namespace Dumux::Properties
 
 #endif