Fixed compile errors in previous commit