Totally forgot to add this fix in before merging, whoops!