Fix broken migration introduced in 2.2.0-rc.1