Fixed broken main