Fix accidentally committed broken code