Do we need to extend this method too to check the new association?