I was missing return statements :(