Fix for test_callback
Merge pull request #11 from NielsZeilemaker/unit_test_fixing_2

Fix for test_callback