fix test
refacotoring