improved coverage

improved coverage