Fix #4 bug.