Fixes work grid bug