Strange workaround to fix a bug with NVIDIA gpu