fixed errors from tests

fixed errors from tests