Now properly stubbing the method call