Update build script for test dir