Another small fix for compatibility with QUDA 0.5.0