fix imports per #537