Fix platform upgrade tests