Updated test and attempting to mock out call