Fix the build again, removed a bit too much #4