A quick fix for #39.

Closes #39.