Fix driver for new kernels