forgot to add the fix to get old tests working again