fix for issue4

fix for #4