Fix code formatting issue