fix puneet's problem