Fixed tests. My bad.