Add specs to get back to 100% coverage