Update to use the correct extension method