diff --git a/chia/full_node/mempool_check_conditions.py b/chia/full_node/mempool_check_conditions.py
index 0fc1da64a3f6..f1905364d273 100644
--- a/chia/full_node/mempool_check_conditions.py
+++ b/chia/full_node/mempool_check_conditions.py
@@ -3,7 +3,7 @@
 import logging
 from typing import Dict, List, Optional
 
-from chia_rs import ALLOW_BACKREFS, DISALLOW_INFINITY_G1, ENABLE_BLS_OPS_OUTSIDE_GUARD, ENABLE_FIXED_DIV, MEMPOOL_MODE
+from chia_rs import MEMPOOL_MODE, get_flags_for_height_and_constants
 from chia_rs import get_puzzle_and_solution_for_coin as get_puzzle_and_solution_for_coin_rust
 from chia_rs import run_block_generator, run_block_generator2, run_chia_program
 
@@ -30,15 +30,6 @@
 log = logging.getLogger(__name__)
 
 
-def get_flags_for_height_and_constants(height: int, constants: ConsensusConstants) -> int:
-    flags = ENABLE_BLS_OPS_OUTSIDE_GUARD | ENABLE_FIXED_DIV | ALLOW_BACKREFS
-
-    if height >= constants.SOFT_FORK5_HEIGHT:
-        flags = flags | DISALLOW_INFINITY_G1
-
-    return flags
-
-
 def get_name_puzzle_conditions(
     generator: BlockGenerator,
     max_cost: int,