Update tests for new form of FlappCommand.run