Set drive current