Revert "fixes broken create visualisation"

This reverts commit 7817e7c5f7bd945d6b32af89089f7b4e32482016.

Thought I pushed to my forked repo. SORRY!