small fix for main