Fixed commented-out code issue