Made the same change to the implementation file as well