Fixed not-so-compiling code