Remove keys from test set