Add real test coverage