diff --git a/test_suite/README.md b/test_suite/README.md
new file mode 100644
index 00000000..39b6f872
--- /dev/null
+++ b/test_suite/README.md
@@ -0,0 +1,18 @@
+#### To run unit tests
+
+```sh
+cargo test
+```
+
+#### To run ui tests
+
+```sh
+(cd deps && cargo clean && cargo update && cargo build)
+cargo test --features compiletest
+```
+
+#### To update goldens after running ui tests
+
+```sh
+tests/ui/update-references.sh
+```
diff --git a/test_suite/tests/compiletest.rs b/test_suite/tests/compiletest.rs
index 385ddc5c..1b70621f 100644
--- a/test_suite/tests/compiletest.rs
+++ b/test_suite/tests/compiletest.rs
@@ -15,6 +15,7 @@ fn ui() {
              --extern serde_derive \
              ",
         )),
+        build_base: std::path::PathBuf::from("../target/ui"),
         ..Default::default()
     });
 }
diff --git a/test_suite/tests/ui/update-references.sh b/test_suite/tests/ui/update-references.sh
index aa99d35f..72594329 100755
--- a/test_suite/tests/ui/update-references.sh
+++ b/test_suite/tests/ui/update-references.sh
@@ -19,32 +19,20 @@
 # If you find yourself manually editing a foo.stderr file, you're
 # doing it wrong.
 
-if [[ "$1" == "--help" || "$1" == "-h" || "$1" == "" || "$2" == "" ]]; then
-    echo "usage: $0 <build-directory> <relative-path-to-rs-files>"
-    echo ""
-    echo "For example:"
-    echo "   $0 ../../../build/x86_64-apple-darwin/test/ui *.rs */*.rs"
-fi
+cd "$(dirname "${BASH_SOURCE[0]}")"
+BUILD_DIR="../../../target/ui"
 
-MYDIR=$(dirname $0)
-
-BUILD_DIR="$1"
-shift
-
-while [[ "$1" != "" ]]; do
-    STDERR_NAME="${1/%.rs/.stderr}"
-    STDOUT_NAME="${1/%.rs/.stdout}"
-    shift
-    if [ -f $BUILD_DIR/$STDOUT_NAME ] && \
-           ! (diff $BUILD_DIR/$STDOUT_NAME $MYDIR/$STDOUT_NAME >& /dev/null); then
-        echo updating $MYDIR/$STDOUT_NAME
-        cp $BUILD_DIR/$STDOUT_NAME $MYDIR/$STDOUT_NAME
+for testcase in */*.rs; do
+    STDERR_NAME="${testcase/%.rs/.stderr}"
+    STDOUT_NAME="${testcase/%.rs/.stdout}"
+    if [ -f "$BUILD_DIR/$STDOUT_NAME" ] && \
+           ! (diff "$BUILD_DIR/$STDOUT_NAME" "$STDOUT_NAME" >& /dev/null); then
+        echo "updating $STDOUT_NAME"
+        cp "$BUILD_DIR/$STDOUT_NAME" "$STDOUT_NAME"
     fi
-    if [ -f $BUILD_DIR/$STDERR_NAME ] && \
-           ! (diff $BUILD_DIR/$STDERR_NAME $MYDIR/$STDERR_NAME >& /dev/null); then
-        echo updating $MYDIR/$STDERR_NAME
-        cp $BUILD_DIR/$STDERR_NAME $MYDIR/$STDERR_NAME
+    if [ -f "$BUILD_DIR/$STDERR_NAME" ] && \
+           ! (diff "$BUILD_DIR/$STDERR_NAME" "$STDERR_NAME" >& /dev/null); then
+        echo "updating $STDERR_NAME"
+        cp "$BUILD_DIR/$STDERR_NAME" "$STDERR_NAME"
     fi
 done
-
-