Change tests back