Fix main build

Signed-off-by: Alex Jackson <ajackson@pivotallabs.com>