Small API change