Small change to fix mistake