Bug fix, need to call the function.