more fixes

Signed-off-by: Yaki Khadafi <elsoldollo@gmail.com>