update whitebox test code