Update to use the correct type specification.