Include upgrade step for issue_201