Fixed bug for button visualisation