Before: <img width="340" height="265" alt="Image" src="https://github.com/user-attachments/assets/5e8a64ab-a6c0-4e10-aad2-50563f12c236" /> After: <img width="339" height="258" alt="Image" src="https://github.com/user-attachments/assets/af8669ed-d20f-4a8e-850c-a808bd6be259" /> Desired result upon setting the return type <img width="448" height="251" alt="Image" src="https://github.com/user-attachments/assets/6465d01d-e9b2-45bf-9fb5-e434a452a5ae" />
Before:
After:
Desired result upon setting the return type