Function sometimes doesn't get drawn. Fixing this.
Merge pull request #36 from barmstro/fix-wat-step

Function sometimes doesn't get drawn. Fixing this.