Small update to fuse layer