Quick fix to get it working again!