align to api change
small cleanup