Fix test
Replace anonumous class with inner lambda