Fixed small method not found bug