Fixes uneven boxes problem in visualisation