I am using Bricks Builder as my page builder and theme. I found a ‘hack’ / work around for this. I added the short code to the block editor for each page I want it to load on and its working now.
Maybe test this and add this to the documentation, I’m sure their will be the same issue with other page builders.