Reverting test code change