diff --git a/shippable.yml b/shippable.yml
index 39ce1f9..82c5461 100644
--- a/shippable.yml
+++ b/shippable.yml
@@ -1,5 +1,7 @@
 language: python
 
+build_image: drydock/u14pytpls:prod
+
 python:
    - 2.7
    - 2.6
@@ -10,16 +12,14 @@ install:
 # Postgres binds to 127.0.0.1 by default and is started on boot. Default username is "postgres" with no password
 # Create a DB as part of before script to use it
 
-
+services:
+  - postgres
 before_script:
   - psql -c 'drop database if exists test;' -U postgres -w
   - psql -c "create database test;" -U postgres
   - mkdir -p shippable/testresults
   - mkdir -p shippable/codecoverage
 
-
-
-
 script:
   - nosetests test.py --with-xunit --xunit-file=shippable/testresults/nosetests.xml
   - which python && coverage run --branch test.py
@@ -29,9 +29,10 @@ script:
 after_script:
   - cp README.md ./shippable
 archive: true
+
 notifications:
      email:
          recipients:
              - buildsampletest@gmail.com
-         on_success: always
+         on_success: success
          on_failure: always
diff --git a/test.py b/test.py
index f076672..df9deba 100644
--- a/test.py
+++ b/test.py
@@ -8,7 +8,7 @@ def test_db(self):
         pg = Postgres()
         pg.populate()
         count = pg.read()
-        self.failIf(count != 4)
+        self.failIf(count != 5)
         pg.disconnect()