Fixed the shutdown in Driver