Fix test to pass if bug is fixed