Fixed an infinite loop... whoops.