Modify test to 95%coverage