small fix so it works with all drivers