unit testing

Get unit testing working again!