minor change

still don't work, need to fix this.