fixed minor bug with issue 19 fix