Minor adjustment to the aoe script