revise test to increse coverage