Minor edit to unit test

[ci skip]