fix to bug