Minor fix for release