Oops, fix error from last commit