diff --git a/test/mock/mock_IBattleInfoCallback.h b/test/mock/mock_IBattleInfoCallback.h
index 63bfa2a9e..3a61110b9 100644
--- a/test/mock/mock_IBattleInfoCallback.h
+++ b/test/mock/mock_IBattleInfoCallback.h
@@ -32,14 +32,14 @@ public:
 	MOCK_CONST_METHOD1(battleGetUnitsIf, battle::Units(const battle::UnitFilter &));
 
 	MOCK_CONST_METHOD1(battleGetUnitByID, const battle::Unit *(uint32_t));
-	MOCK_CONST_METHOD2(battleGetUnitByPos, const battle::Unit *(BattleHex, bool));
+	MOCK_CONST_METHOD2(battleGetUnitByPos, const battle::Unit *(const BattleHex&, bool));
 	MOCK_CONST_METHOD0(battleActiveUnit, const battle::Unit *());
 
 	MOCK_CONST_METHOD0(getBonusBearer, IBonusBearer*());
 	MOCK_CONST_METHOD0(getBattle, IBattleInfo*());
 	MOCK_CONST_METHOD0(getPlayerID, std::optional<PlayerColor>());
 
-	MOCK_CONST_METHOD2(battleGetAllObstaclesOnPos, std::vector<std::shared_ptr<const CObstacleInstance>>(BattleHex, bool));
+	MOCK_CONST_METHOD2(battleGetAllObstaclesOnPos, std::vector<std::shared_ptr<const CObstacleInstance>>(const BattleHex&, bool));
 	MOCK_CONST_METHOD2(getAllAffectedObstaclesByStack, std::vector<std::shared_ptr<const CObstacleInstance>>(const battle::Unit *, const BattleHexArray &));
 
 };