Update test to reflect results from mock