Small tweak to the code formatting