diff --git a/scripts/common.make b/scripts/common.make
index 9908a3c..cafad26 100644
--- a/scripts/common.make
+++ b/scripts/common.make
@@ -22,7 +22,7 @@
 LDFLAGS ?=
 
 LOCAL_CFLAGS = $(WARNING_CFLAGS) -I$(MBEDTLS_TEST_PATH)/include -I$(MBEDTLS_PATH)/include -D_FILE_OFFSET_BITS=64
-LOCAL_CXXFLAGS = $(WARNING_CXXFLAGS) -I$(MBEDTLS_PATH)/include -I$(MBEDTLS_PATH)/tests/include -D_FILE_OFFSET_BITS=64
+LOCAL_CXXFLAGS = $(WARNING_CXXFLAGS) -I$(MBEDTLS_PATH)/include -I$(MBEDTLS_PATH)/framework/include -D_FILE_OFFSET_BITS=64
 LOCAL_LDFLAGS = ${MBEDTLS_TEST_OBJS} 		\
 		-L$(MBEDTLS_PATH)/library			\
 		-lmbedtls$(SHARED_SUFFIX)	\
diff --git a/scripts/generate_visualc_files.pl b/scripts/generate_visualc_files.pl
index d0d18a3..b57d7e3 100755
--- a/scripts/generate_visualc_files.pl
+++ b/scripts/generate_visualc_files.pl
@@ -26,8 +26,8 @@
 my $psa_header_dir = 'include/psa';
 my $source_dir = 'library';
 my $test_source_dir = 'framework/src';
-my $test_header_dir = 'tests/include/test';
-my $test_drivers_header_dir = 'tests/include/test/drivers';
+my $test_header_dir = 'framework/include/test';
+my $test_drivers_header_dir = 'framework/include/test/drivers';
 my $test_drivers_source_dir = 'framework/src/drivers';
 
 my @thirdparty_header_dirs = qw(
@@ -48,7 +48,7 @@
     3rdparty/everest/include/everest
     3rdparty/everest/include/everest/vs2013
     3rdparty/everest/include/everest/kremlib
-    tests/include
+    framework/include
 );
 my $include_directories = join(';', map {"../../$_"} @include_directories);