Code formatting fix