Fixed a unit test broken by last change

Signed-off-by: Ozan Turgut <ozanturgut@gmail.com>