Fix import error with prettytable