Added if statements around the offending BABY code.