Fixed #157  thanks!