Fix STS import issue