Fix remove from array test