fix errors

lib/pages/package/package_info.dart

@@ -25,10 +25,12 @@ import 'package:provider/provider.dart';
 final DateFormat dateFormat = DateFormat("d MMM yyyy");
 
 class PackageInfo extends StatefulWidget {
+  final isCustomer;
   final isSearchResult;
   final Package package;
 
-  PackageInfo({this.package, this.isSearchResult = false});
+  PackageInfo(
+      {this.package, this.isSearchResult = false, this.isCustomer = false});
 
   @override
   _PackageInfoState createState() => _PackageInfoState();
@@ -156,7 +158,8 @@ class _PackageInfoState extends State<PackageInfo> {
                   _package.photoUrls.length == 0 ? Container() : img,
                   widget.isSearchResult ? Container() : descBox,
                   remarkBox,
-                  _package.status == package_processed_status
+                  _package.status == package_received_status &&
+                          widget.isCustomer
                       ? returnButton
                       : Container(),
                   widget.isSearchResult ? Container() : deliveryAddressBox,