very dirty fix bug for issue #2 thanks @jrhorn424