Minor patch of function signature that shouldn't have been pushed