don't flatten in test