Fix test to use correct key_val_parameter