improve test to add more coverage