maybe move that into a method to return the dict, make tests shorter