Quick fix to previous commit