Fix STS import