#9 update tests