Fix type error in main function

Signed-off-by: Momchil Atanasov <m.momchil@gmail.com>