diff --git a/mitogen/profiler.py b/mitogen/profiler.py index bbf6086a..512a593e 100644 --- a/mitogen/profiler.py +++ b/mitogen/profiler.py @@ -90,7 +90,7 @@ def merge_stats(outpath, inpaths): break time.sleep(0.2) - stats.dump_stats(outpath) + pstats.dump_stats(outpath) def generate_stats(outpath, tmpdir): diff --git a/tests/testlib.py b/tests/testlib.py index 019b35d7..d40ce573 100644 --- a/tests/testlib.py +++ b/tests/testlib.py @@ -144,7 +144,7 @@ def wait_for_port( if not pattern: # Success: We connected & there's no banner check to perform. - sock.shutdown(socket.SHUTD_RDWR) + sock.shutdown(socket.SHUT_RDWR) sock.close() return