Using Air Mouse

My Air Mouse connects to Apple TV but only mouse element works. When I start to use virtual keyboard, it throws me back into the old and painful aTV keyboard. Help! The virtual keyboard did originally work for about 5 minutes!

I am using an Apple iPhone 3G