Fixes visualization