Main.cpp fix

Fixes a bug in the main so compilation is posible