closed #18 small fix