Fixes some forgotten tests