Fixed more kmalloc bugs