BUG: Fixed a line I had commented out for testing and forgot to renable.