Cleanup minor change