Fix array edge cases