1 files changed, 12 insertions, 2 deletions
diff --git a/test_pbuilder-modules b/test_pbuilder-modules
index 402c424..bec24e1 100755
--- a/test_pbuilder-modules
+++ b/test_pbuilder-modules
@@ -1,7 +1,17 @@
 #!/bin/bash
 
-. ./testlib.sh
-. ./pbuilder-modules
+if [ -n "$PBUILDER_CHECKOUT" ]; then
+    . "$PBUILDER_CHECKOUT/testlib.sh"
+
+    . "$PBUILDER_CHECKOUT/pbuilder-modules"
+else
+    # these currently don't need to be exported
+    PBUILDER_TEST_ROOT="${PBUILDER_ROOT:-}"
+    PBUILDER_TEST_PKGLIBDIR="${PBUILDER_PKGLIBDIR:-$PBUILDER_ROOT/usr/lib/pbuilder}"
+    . "$PBUILDER_TEST_PKGLIBDIR/testlib.sh"
+
+    . "$PBUILDER_TEST_PKGLIBDIR/pbuilder-modules"
+fi
 
 function test_information() {
     log "I: test"