Get unit tests working
Remove debug call