Update stress to use driver 2.0.4