Small change to fix compile errors