Fix test-unit