Set STB enable in main test code