quick fix to bug