Add back accidentally deleted code from d00460ab5f041ca47ec4145bffd4857cab008bb0
Add back accidentally deleted code from d00460ab5f041ca47ec4145bffd4857cab008bb0

DMOJ/site,DMOJ/site,DMOJ/site,DMOJ/site