Fix failing IT

Signed-off-by: Alin Dreghiciu <adreghiciu@gmail.com>