Fix array bug