Hopefully fixes tests, can't run them locally :(