Added missing test code.