Fixed failing test

Forgot to add new dependencies