Minor bug fix and version update