Add test for STI referencing