Make code style in test consistent