Fix corresponding unit test

Had already pushed the change, so I couldn't amend it!