Driver Register fix (remain bug)