TEST: fix unittest