fix end in viz controller