Fix torch overlay

Signed-off-by: iceandfire <arhamjamal@gmail.com>