Adding wait for driver