fix code changes

Signed-off-by: Yaniv Bronhaim <ybronhei@redhat.com>
Merge pull request #9 from ficoos/wrong-code

fix code changes