Small Fix Samy's visualization