Fixed issue #18

Issue #18 seems to be fixed. I just tested it on AMD GPU an it works, the samething on two nVidia GPUs.