Adding missing dependency for running the tests

Forgot to add this dependency on last commit