Small code formatting fix