Fixes for the compute shader