Should fix the error @toddkaufman saw

(totally untested no idea if it works)