Small modification to match the new signature.