Fix error by missing functions from module