Fix comment out bug