Fix #6 bug.