Improve signal handling to prevent signals from being lost.