Bugfix - Incorrect visualisation