Fixed compilation errors. Oops :P