Fix wrong method logic