Fix tests. Forgot to remove this on my last commit. Whoops