Fix tests

- Update requirements.txt