Fix bug with calling 'handle_drain' connection method when writes are unblocked.