Remove assert import