[BUG] fixes #3

fixes #3