Fix deprecated import that was causing warnings on startup