fixing a non working test + improving coverage