Undefined var notice fix.