Add test spec to check coverage