Fixed code so it would compile