Fix compile errors on tests

Fix of #2.