Fixed commented out code