Problem statement fix