US536 - Fix dodgy import