small fix for visualizing