fix failed for test
minor change