Change behaviour of expect.keys function