Fix import for ST2