quick fix for main