Some kmemleak fixes