Fix augmented to return the implementation