Fixed broken tests after changing Piece model structure