Fixing the extension package integration