Bugfix 2

I'm terrible at this