Fixed it so it compiles