Temporary fix for #9

Signed-off-by: Anurag Priyam <anurag08priyam@gmail.com>