Quick fix: Fixing node element selection