PROBCORE-402 Fixed bug in visualization