remove some warnings in custom.

Signed-off-by: Daogang Tang <daogangtang@gmail.com>