fixes #5

closes #5

Signed-off-by: Mohamad Ahmadi <mhahmadi@nteropcorp.com>