Fix code formatting introduced by PR #2333

Fixing my previous PR which messed up the code formatting.