Fixing code formatting error