Fixed minor warning in cube_driver.c