slight fix to the previous commit