Fixed non-compiling code. GJ Magnus.