one last tiny change

got the toggle to work by changing the return type