Update test for PrivateEc2 driver