missing include
remove old logic that was breaking clean shutdown