Small change due to comment on github