In your Settings, Display & Brightness - amend your Auto-Lock to 1 minute or above