oops, fixed minor bug