Fix: return the right callbacks for deviceSetup / teardown