fix #2

Fixes #2