Updated for compile.
Mini tweak.