add test spec and make it pass