Adding test coverage for single generator units