fixing small compiation error in main