docs tests: Fix use of migrate_set_parameter