Fix mistake from last commit