slight fix to previous change