State-aware modeling