missed an endif
Merge pull request #3 from yamatt/master

missed an endif