fix test & got rid of flatten in args