Fix code formatting introduced by PR #2333

Fixing my previous PR (#2333) which messed up the code formatting.