fix to bug in previous commit