fixes #11 Thanks!

thanks @baringji