When loading, restore the right visualization