Change test so it'll pass

TODO rebase (squash)