TST: Adding coverage