fixing tests

third time's a charm!