Fixed code to compile