adapted unittest to fix