Fix #7

Fixes #7