Added missing unittest