Minor bug fix and version bump