I should update the return type.