Fixes #4

fix