TST: Fix broken test and remove assert call