put key_id method back