fixed tests so they now pass