Quick fix to a bug.