Ok, mockito is much better than my stub methods, should have known this before :)