Fixes diff unit test; proposed by @n0rthwood