fixing testing issue