v2.0.1

Minor fix.