fixed build fail due to wrong package import