Update the test script regarding 'CDE-316'
Update script to be more accuracy.