updated tests after implements change