Minor fix

Two simple fixes.