diff --git a/pg_backup.sh b/pg_backup.sh
index 84ab0b1..4029803 100644
--- a/pg_backup.sh
+++ b/pg_backup.sh
@@ -1,16 +1,15 @@
-#!/bin/sh
+#!/bin/bash
 
 set -e
 
 BACKUP_FILE='/var/lib/postgresql/data/backup.sql'
 
-backup() {
-  PGPASSWORD=$(cat "$POSTGRES_PASSWORD_FILE")
-  export PGPASSWORD
-  pg_dump -U "${POSTGRES_USER}" "${POSTGRES_DB}" > $BACKUP_FILE
+function backup {
+  export PGPASSWORD=$(cat $POSTGRES_PASSWORD_FILE)
+  pg_dump -U ${POSTGRES_USER} ${POSTGRES_DB} > $BACKUP_FILE
 }
 
-restore() {
+function restore {
     cd /var/lib/postgresql/data/
     restore_config(){
         # Restore allowed connections
@@ -24,12 +23,12 @@ restore() {
     trap restore_config EXIT INT TERM
 
     # Recreate Database
-    psql -U "${POSTGRES_USER}" -d postgres -c "DROP DATABASE ${POSTGRES_DB} WITH (FORCE);" 
-    createdb -U "${POSTGRES_USER}" "${POSTGRES_DB}"
-    psql -U "${POSTGRES_USER}" -d "${POSTGRES_DB}" -1 -f $BACKUP_FILE
+    psql -U ${POSTGRES_USER} -d postgres -c "DROP DATABASE ${POSTGRES_DB} WITH (FORCE);" 
+    createdb -U ${POSTGRES_USER} ${POSTGRES_DB}
+    psql -U ${POSTGRES_USER} -d ${POSTGRES_DB} -1 -f $BACKUP_FILE
 
     trap - EXIT INT TERM
     restore_config
 }
 
-"$@"
+$@