Minor adjustment

Minor adjustment