Fixed driver to use correct type.