Bug fix. Need to figure out how to test locally, haha.