Fixing broken test dependency