Small undefined error fix.