Good idea, I've added the fix