Although things are working i can't seem to write a proper test to check if methods are being called correctly.