Update test. Fix bug.