quick_fix
Fixed small bug that broke the code. Whoops.