fix tests around get_assignment