fix for issue #139 - sebastianbergmann#139

Fix issue with mocking for traits. Current code mock all existen methods what is wrong.