minor change on test code