Fix method declaration issue