Fix array, add test