Fix #2

Fixes #2 (hopefully).