Creating unit tests for damage control