Fix test code style