fix build warnings

Fix build warnings caused by previous fix.
Fix warning of vmalloc

Signed-off-by: Shlomo Pongratz <shlomop@mellanox.com>