Quick fix for broken code