The fact we need to do this when testing is kinda hacky.