Small unit test fix
Fixed #728