Fix broken tests form last commit