Tiny fix to the previous commit