diff --git a/dropshots/src/main/java/com/dropbox/dropshots/Dropshots.kt b/dropshots/src/main/java/com/dropbox/dropshots/Dropshots.kt
index 81c7de6..50efc5e 100644
--- a/dropshots/src/main/java/com/dropbox/dropshots/Dropshots.kt
+++ b/dropshots/src/main/java/com/dropbox/dropshots/Dropshots.kt
@@ -272,7 +272,7 @@ public class Dropshots @JvmOverloads public constructor(
 }
 
 internal fun defaultRootScreenshotDirectory(): File {
-  val context = InstrumentationRegistry.getInstrumentation().targetContext
+  val context = InstrumentationRegistry.getInstrumentation().context
   val externalStorageDir = Environment
     .getExternalStoragePublicDirectory(Environment.DIRECTORY_DOWNLOADS)
   return File(externalStorageDir, "screenshots/${context.packageName}")
diff --git a/tests/src/androidTest/assets/MatchesActivityScreenshot.png b/tests/src/androidTest/assets/MatchesActivityScreenshot.png
index b6ec16c..71090f1 100644
Binary files a/tests/src/androidTest/assets/MatchesActivityScreenshot.png and b/tests/src/androidTest/assets/MatchesActivityScreenshot.png differ
diff --git a/tests/src/androidTest/assets/MatchesFullScreenshot.png b/tests/src/androidTest/assets/MatchesFullScreenshot.png
index a37c75c..8aa629d 100644
Binary files a/tests/src/androidTest/assets/MatchesFullScreenshot.png and b/tests/src/androidTest/assets/MatchesFullScreenshot.png differ
diff --git a/tests/src/androidTest/assets/MatchesViewScreenshot.png b/tests/src/androidTest/assets/MatchesViewScreenshot.png
index 415c49c..2dda90b 100644
Binary files a/tests/src/androidTest/assets/MatchesViewScreenshot.png and b/tests/src/androidTest/assets/MatchesViewScreenshot.png differ