Removed unnecessary set, although tests still worked with it in :)