Fix type in test