I've worked out what the problem was. I'm using the Divi Theme Builder - there is an option in there for the editor. I had the classic editor selected. Toolset seems to ignore that setting in the Custom Templates section but not in the archive section. Setting it to Gutenberg fixed the problem