Minor fix to get rid of warnings