Fix so it works on my GPU too