Little fix for the Main