Added missing dependency that was making the tests not pass