Another tests fix