Make a little less hardcoded