Whoops, fixed a bug