Fix sth. Can run the test now.