Fix error with test_issue_256