Small fix (Fixes #1)