Fixed some broken code.