Fix forgotten method