Add done callbacks to get websockets running again