Fixes error on visualization stats