Quick fix to package main