Fixed tests, which were broken but didn't break the build (wtf)