Fixed drive_func's conditional