modify to fix package dependency