fix code changes

Signed-off-by: Yaniv Bronhaim <ybronhei@redhat.com>