Fixes upgrade step for issue #218