fixes

Signed-off-by: Raeef Refai <raeef.refai@gmail.com>