Ah that makes sense. Removing `get_current_screen()` and adding `is_admin()` check seem to address all edge cases now.