Fixes during testing
whoops