fixing test changes from last commit