[OOPS] fixing some tests I just broke