Fixed compile errors

Fixed compile errors