Fix test dependency