Add missing method
fix copy pasta