Good idea. I just fixed it with your code.