Fix previous commit

Signed-off-by: Andrey Shamis <andrey.shamis@intel.com>