Add some unittest replacements