Fixed package reference issue