Fixed Piece to include getTeam method