fixing tests based on new changes