Fixed the function signature