fixed some broken tests as a result of some incorrect stubbed classes