[fix] Fixed tests broken by e35d98b75

Kids, always run your tests before pushing to master