Removed extra ui ref from code base