diff --git a/MuonSpectrometer/MuonConfig/python/MuonGeometryConfig.py b/MuonSpectrometer/MuonConfig/python/MuonGeometryConfig.py
index c500e303669ff498435cd636cc06a720421c20f7..6202cee2ea6e8b1fd81e6dd46da6e435c9ea056a 100644
--- a/MuonSpectrometer/MuonConfig/python/MuonGeometryConfig.py
+++ b/MuonSpectrometer/MuonConfig/python/MuonGeometryConfig.py
@@ -1,8 +1,9 @@
-# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
+# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
 from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
 from AthenaConfiguration.ComponentFactory import CompFactory
 from AthenaConfiguration.AccumulatorCache import AccumulatorCache
+from AthenaConfiguration.Enums import ProductionStep
 from AtlasGeoModel.GeoModelConfig import GeoModelCfg
 from AthenaConfiguration.Enums import LHCPeriod
 
diff --git a/Tools/WorkflowTestRunner/python/References.py b/Tools/WorkflowTestRunner/python/References.py
index e30ccf69c5b066dd66655dee211d8385ffcd10bd..3d3ac84612f91971352a9200e11be2e072411da4 100644
--- a/Tools/WorkflowTestRunner/python/References.py
+++ b/Tools/WorkflowTestRunner/python/References.py
@@ -32,5 +32,5 @@ references_map = {
     "data_PHYS_Run3": "v16",
     "mc_PHYS_Run2": "v21",
     "mc_PHYS_Run3": "v21",
-    "af3_PHYS_Run3": "v1",
+    "af3_PHYS_Run3": "v2",
 }