Fixed compile errors...