Minor fix for warnings