slight fix to my previous commit