Make dx run use the stage_instance_types arg and fix tests