update tests
fix silly mistake