Update to support extension.