Closes #4 bug.