temporary fix on error from refactor