Reverting test fix