Fix CustomScriptExtension tests