Fixed unittest failure on old tests.