mini edit to tests