Reverting test changes