Add fixes to arena system script