improve testing coverage