fix return key to be array