Fix formatting in tests