Fixing dependency issue

Signed-off-by: Alan Gardner <alan@armoin.com>