Fix tests

Change-Id: Ifbf82076f1dee4c16127944eacf83cca2490f44d
Signed-off-by: Bogdan Dobrelya <bdobrelia@mirantis.com>