Fix test code indentation