Fixed something getting lost in between commits