fix #1

thanks @blakeembrey