Fix #8 bug.