Fix formatting of if statement in test code

Fix formatting of if statement in test code #2381