Make sure element_to_count_dict passes on arguments

actually fixes #23