Add missing dependency in migration