Fixed the code so that it compiles... at least.