Opps, fix bug