Fixed mockito problem and made a real test!