Fix my idiocy from latest commit