fix to bug left in previous commit