fix forgotten method from refacotoring