update the test according to Kenneth's comments