Fix tests on contest mode