Fix tests.

Need to reset the limited attribute once it is set to True.