Fix assignment errors after method check refactoring