Minor driver update