Fixing the build. Added missing method