diff --git a/test_app/CMakeLists.txt b/test_app/CMakeLists.txt
index d9e249f..8e9a679 100644
--- a/test_app/CMakeLists.txt
+++ b/test_app/CMakeLists.txt
@@ -2,9 +2,10 @@
 # CMakeLists in this exact order for cmake to work correctly
 cmake_minimum_required(VERSION 3.5)
 
-set(EXTRA_COMPONENT_DIRS "../libsodium")
+set(EXTRA_COMPONENT_DIRS ../libsodium ../expat)
 
 # Set the components to include the tests for.
-set(TEST_COMPONENTS libsodium CACHE STRING "List of components to test")
+set(TEST_COMPONENTS libsodium expat CACHE STRING "List of components to test")
+
 include($ENV{IDF_PATH}/tools/cmake/project.cmake)
 project(idf_extra_test_app)
diff --git a/test_app/partitions.csv b/test_app/partitions.csv
new file mode 100644
index 0000000..75c7787
--- /dev/null
+++ b/test_app/partitions.csv
@@ -0,0 +1,5 @@
+# Name,   Type, SubType, Offset,  Size, Flags
+# Note: if you have increased the bootloader size, make sure to update the offsets to avoid overlap
+nvs,      data, nvs,     ,        0x6000,
+phy_init, data, phy,     ,        0x1000,
+factory,  app,  factory, ,        2M,
diff --git a/test_app/sdkconfig.defaults b/test_app/sdkconfig.defaults
index 808f92a..9801b6c 100644
--- a/test_app/sdkconfig.defaults
+++ b/test_app/sdkconfig.defaults
@@ -1,2 +1,6 @@
 CONFIG_ESP_INT_WDT=n
 CONFIG_ESP_TASK_WDT=n
+CONFIG_ESPTOOLPY_FLASHSIZE_4MB=y
+CONFIG_PARTITION_TABLE_CUSTOM=y
+CONFIG_PARTITION_TABLE_CUSTOM_FILENAME="partitions.csv"
+CONFIG_PARTITION_TABLE_FILENAME="partitions.csv"