Corrected an error i missed after changes made in last commit