Preparing for STI changes

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