Small driver fix