update test accordingly