fix bug, now working.