[PATCH] NIU: fix missing revert of return and fix the driver