fixed tests to account for additions [MJ-374]