Fixing unit test deps