Looks like you missed adding these tests