* Fix unittest