Added forgetten test code.