Minor fix for 0.0.6