You should update the function arguments below too. We shouldn't rely on other files loading the appropriate functions before this one.