oops, fix bug