diff --git a/samples/dml.js b/samples/dml.js
index 31f3cc1a4..098997c61 100644
--- a/samples/dml.js
+++ b/samples/dml.js
@@ -638,7 +638,6 @@ async function executeSqlWithCustomTimeoutAndRetrySettings(
   // [START spanner_set_custom_timeout_and_retry]
   // Imports the Google Cloud client library
   const {Spanner} = require('@google-cloud/spanner');
-  const {grpc} = require('google-gax');
 
   /**
    * TODO(developer): Uncomment the following lines before running the sample.
@@ -651,10 +650,11 @@ async function executeSqlWithCustomTimeoutAndRetrySettings(
   const spanner = new Spanner({
     projectId: projectId,
   });
-
+  const DEADLINE_EXCEEDED_STATUS_CODE = 4;
+  const UNAVAILABLE_STATUS_CODE = 14;
   const retryAndTimeoutSettings = {
     retry: {
-      retryCodes: [grpc.status.DEADLINE_EXCEEDED, grpc.status.UNAVAILABLE],
+      retryCodes: [DEADLINE_EXCEEDED_STATUS_CODE, UNAVAILABLE_STATUS_CODE],
       backoffSettings: {
         // Configure retry delay settings.
         initialRetryDelayMillis: 500,