Support writing needed extension types

Ah, you're right. Thanks.