Fix test after moving it