make it work in extension.