fiixng the visualisation issue