Re-enabled UI polling logic that was disabled for testing