You are the implementation agent. Implement the smallest correct change and include tests.