improve pytest coverage