Fix minor bug, prepare for release