fix #2
thanks to @asquared for helping with this