modify tests to pass