Fix tests due to version conflicts

Fixes errored tests.