diff --git a/.travis.yml b/.travis.yml
index eb5e392..d2c3948 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,9 +1,13 @@
 services:
 - docker
 
+env:
+  global:
+  - PRESTOSQL_RELEASE=329
+
 install:
-- curl -LO https://github.com/IBM/presto-db2/releases/download/325/presto-db2-325.zip
-- unzip presto-db2-325.zip
+- curl -LO https://github.com/IBM/presto-db2/releases/download/${PRESTOSQL_RELEASE}/presto-db2-${PRESTOSQL_RELEASE}.zip
+- unzip presto-db2-${PRESTOSQL_RELEASE}.zip
 
 before_script:
   curl -LO https://storage.googleapis.com/container-structure-test/latest/container-structure-test-linux-amd64 &&
diff --git a/Dockerfile b/Dockerfile
index 0affae9..be89a13 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM prestosql/presto:325
+FROM prestosql/presto:329
 
 # Add Db2 connector
-COPY --chown=presto:presto presto-db2-325 /usr/lib/presto/plugin/db2
+COPY --chown=presto:presto presto-db2-329 /usr/lib/presto/plugin/db2
diff --git a/test_config.yaml b/test_config.yaml
index 7b95fba..2b063d3 100644
--- a/test_config.yaml
+++ b/test_config.yaml
@@ -8,6 +8,8 @@ fileExistenceTests:
   path: /usr/lib/presto/plugin/db2
   uid: 1000
   gid: 1000
+- name: presto-db2.jar
+  path: /usr/lib/presto/plugin/db2/presto-db2-329.jar
 metadataTest:
   env:
   - key: JAVA_HOME