Base: Fix key processing with real screen state