Updated test to follow new function naming