changes for the unit tests