fix PR feedback