Minor adjustment to an extension method.