Fix other tests I broke