Fix torch QS

Change-Id: I012cc57ea7188df426e6379a2fc9041976261c8a