minor changes to pass TBOT test