Small bug fix, wrong method called