Small bug in my last commit