Fix non-compiling code