fixed bad refactoring on my part... :)