Fix display bug
Merge pull request #4 from aaronlifshin/patch-4

Fix display bug