updating tests
fix broken test