Added an unittest