Fix Unittest