Fixing issue with package import in setup script