Fix little error in tests