Add one more test. Fix BUG.