Fixed infinite loop.... oops.