Whoopsy, reverted some test code