Small undefined variable fix