fix #3 by @nathanial