Added two more tests to get coverage back up