Making minor modifications to test process.