Fix wrong main reference