Fix type in tests