Fix squish tests