FIX: Save the connection state callbacks