Fixed test case bug. Thanks @fkdosilovic