Change to work with new HKT SDK update