Code was wrong, fixed it