Update optional 3.1 migration steps