fix bug
Merge pull request #422 from dimagi/viz

fix bug