Update test for function change