Fixed a bug
Forgot to update this