Fix #5

This fixes #5.