Fix assert call
Merge pull request #4 from sindhus/visualization

Fix assert call