oops, fixed bug