Fix broken source code