prob3 fix