fix unit test

@henar