Fixing test dependency