[BUGFIX] Fix broken unit tests

Fix one test broken since the test case exists and another broken
with commit 870aeee.