Whoops, quick bugfix