array_key_index fix