Properly indent the code now that the test is done