Fixed code so it compiles.