Minor change due to change in implementation