Fix broken code (previous commit)