Fix test, to squash