small efficiency fix