TST: Added two statements for tests that were left out