Add check unit test