Fix CustomKernel-unittest bug.

@karlnapf 
Sorry for my mistake in CustomKernel-unittest.
Now I fix it.