improve need be tested

improve need be tested