Fixed error in previous commit #2