update comment and fix check on