Little fix in code formatting