Fix RosterTest import