Fix for deprecated package