Fixed non-compiling code