Update main.h

small fix