Minor bug fix. Close #91.