Moving B impl to mock