Yes, I foresee an easy fix, but I want to run your test case to make sure there is not something else broken.