small fix to previous commit to correct a crash