Minor fix: code cleanup