working on updating tests