Fixes some forgotten stuff