import sys

main_path = r"C:\Users\reazul.ashraf\workspace\rust\mcp-memory\server\src\main.rs"
with open(main_path, "r", encoding="utf-8") as f:
    content = f.read()

content = content.replace(
    'handler: Arc::new(MemoryHandler { state }),',
    'handler: Arc::new(MemoryHandler { state: Arc::clone(&state) }),'
)

with open(main_path, "w", encoding="utf-8") as f:
    f.write(content)
print("Fixed main.rs move error")