Fix test formatting