mainframe unit tests fix