✅ Add more tests for current behaviour