Fix basic test. Second test not working.