Woops, fix a bug