minor changes due to PR