Fix mistake when making changes